{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "lenet out: torch.Size([2, 10])\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "from torch import nn\n",
    "\n",
    "class LeNet5(nn.Module):\n",
    "    def __init__(self):\n",
    "        super(LeNet5,self).__init__()\n",
    "        self.conv_unit=nn.Sequential(#(b,3,32,32)\n",
    "            nn.Conv2d(3,6,kernel_size=5,stride=1,padding=0),#(b,6,28,28)\n",
    "            nn.MaxPool2d(2,2,padding=0),#(b,6,14,14)\n",
    "            ##nn.Conv2d(输入特征图的深度,输出特征图的深度,kernel_size=长/宽,stride=1,padding=0),\n",
    "            #nn.Conv2d(深            ,组数,         kernel_size=长/宽,stride=1,padding=0),\n",
    "            nn.Conv2d(6,16,kernel_size=5,stride=1,padding=0),#(b,16,10,10)\n",
    "            nn.MaxPool2d(2, 2, padding=0)#(b,16,5,5)\n",
    "        )\n",
    "        self.fc_unit=nn.Sequential(\n",
    "            nn.Linear(16*5*5,120),\n",
    "            nn.Linear(120,84),\n",
    "            nn.Linear(84,10)\n",
    "        )\n",
    "    def forward(self,x):#[样本数2,通道数3,高32,宽32]\n",
    "        m=x.size(0)#样本数量\n",
    "        x=self.conv_unit(x)#[2,16,5,5]\n",
    "        x=x.view(m,16*5*5)#[2,16*5*5]\n",
    "        # x=x.view(-1,32*5*5)\n",
    "        out=self.fc_unit(x)#[2,10]\n",
    "        return out\n",
    "\n",
    "\n",
    "# 定义LeNet5模型\n",
    "model=LeNet5()\n",
    "# 生成一个随机的2x3x32x32的张量\n",
    "tmp=torch.randn(2,3,32,32)\n",
    "# 将张量输入模型，得到输出\n",
    "out=model(tmp)\n",
    "# 打印输出张量的形状\n",
    "print('lenet out:',out.shape)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ..\\..\\DATA\\CIFAR10_data\\cifar-10-python.tar.gz\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|██████████| 170498071/170498071 [2:50:30<00:00, 16664.90it/s]  \n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Extracting ..\\..\\DATA\\CIFAR10_data\\cifar-10-python.tar.gz to ..\\..\\DATA\\CIFAR10_data\n",
      "Files already downloaded and verified\n",
      "x: torch.Size([128, 3, 32, 32]) label: torch.Size([128])\n",
      "Inceptionv1(\n",
      "  (branch1x1): BasicConv2d(\n",
      "    (conv): Conv2d(3, 64, kernel_size=(1, 1), stride=(1, 1))\n",
      "    (bn): BatchNorm2d(64, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "  )\n",
      "  (branch3x3): Sequential(\n",
      "    (0): BasicConv2d(\n",
      "      (conv): Conv2d(3, 96, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(96, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(96, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))\n",
      "      (bn): BatchNorm2d(128, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      "  (brach5x5): Sequential(\n",
      "    (0): BasicConv2d(\n",
      "      (conv): Conv2d(3, 16, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(16, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(16, 32, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))\n",
      "      (bn): BatchNorm2d(32, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      "  (brach_pool): Sequential(\n",
      "    (0): MaxPool2d(kernel_size=3, stride=1, padding=1, dilation=1, ceil_mode=False)\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(3, 32, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(32, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      "  (outlayer): Linear(in_features=352, out_features=10, bias=True)\n",
      ")\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "0 loss: 1.7650731801986694\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "0 test acc: 0.3357\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "1 loss: 1.5748202800750732\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "1 test acc: 0.3614\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "2 loss: 1.6914573907852173\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "2 test acc: 0.3951\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "3 loss: 1.6759464740753174\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "3 test acc: 0.4096\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "4 loss: 1.4922606945037842\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "4 test acc: 0.4381\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "5 loss: 1.580217957496643\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "5 test acc: 0.435\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "6 loss: 1.531269907951355\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "6 test acc: 0.4513\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "7 loss: 1.4775112867355347\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "7 test acc: 0.4557\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "8 loss: 1.3715345859527588\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "8 test acc: 0.4674\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([80, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([80, 352])\n",
      "_______________\n",
      "9 loss: 1.292141318321228\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([128, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([128, 352])\n",
      "_______________\n",
      "torch.Size([16, 352, 1, 1])\n",
      "_______________\n",
      "torch.Size([16, 352])\n",
      "_______________\n",
      "9 test acc: 0.4691\n"
     ]
    }
   ],
   "source": [
    "import  torch\n",
    "from    torch.utils.data import DataLoader\n",
    "from    torchvision import datasets\n",
    "from    torchvision import transforms\n",
    "from    torch import nn, optim\n",
    "\n",
    "def main():\n",
    "    # 定义batch大小\n",
    "    batchsz = 128\n",
    "\n",
    "    # 加载训练集数据\n",
    "    cifar_train = datasets.CIFAR10('..\\..\\DATA\\CIFAR10_data', True, transform=transforms.Compose([\n",
    "        transforms.Resize((32, 32)),\n",
    "        transforms.ToTensor(),\n",
    "        transforms.Normalize(mean=[0.485, 0.456, 0.406],\n",
    "                             std=[0.229, 0.224, 0.225])\n",
    "    ]), download=True)\n",
    "    # 将训练集数据转换为DataLoader格式\n",
    "    cifar_train = DataLoader(cifar_train, batch_size=batchsz, shuffle=True)\n",
    "\n",
    "    # 加载测试集数据\n",
    "    cifar_test = datasets.CIFAR10('..\\..\\DATA\\CIFAR10_data', False, transform=transforms.Compose([\n",
    "        transforms.Resize((32, 32)),\n",
    "        transforms.ToTensor(),\n",
    "        transforms.Normalize(mean=[0.485, 0.456, 0.406],\n",
    "                             std=[0.229, 0.224, 0.225])\n",
    "    ]), download=True)\n",
    "    # 将测试集数据转换为DataLoader格式\n",
    "    cifar_test = DataLoader(cifar_test, batch_size=batchsz, shuffle=True)\n",
    "\n",
    "\n",
    "    # 获取训练集的第一个batch数据\n",
    "    x, label = next(iter(cifar_train))\n",
    "    # 打印数据形状\n",
    "    print('x:', x.shape, 'label:', label.shape)\n",
    "\n",
    "    # 判断是否有可用的GPU，如果没有则使用CPU\n",
    "    # device = torch.device('cuda')\n",
    "    device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
    "\n",
    "    # 加载模型，并移动到指定设备\n",
    "    # model = LeNet5().to(device)\n",
    "    # model = ResNet18().to(device)\n",
    "    model = Inceptionv1().to(device)\n",
    "    ####################################################################################################\n",
    "    # model = ResNet18()\n",
    "\n",
    "    # 定义损失函数\n",
    "    criteon = nn.CrossEntropyLoss().to(device)\n",
    "    # 定义优化器\n",
    "    optimizer = optim.Adam(model.parameters(), lr=1e-3)\n",
    "    # 打印模型结构\n",
    "    print(model)\n",
    "\n",
    "    # 训练模型\n",
    "    for epoch in range(10):\n",
    "\n",
    "        # 设置模型为训练模式\n",
    "        model.train()\n",
    "        # 遍历训练集\n",
    "        for batchidx, (x, label) in enumerate(cifar_train):\n",
    "            # [b, 3, 32, 32]\n",
    "            # [b]\n",
    "            x, label = x.to(device), label.to(device)\n",
    "\n",
    "\n",
    "            # 前向传播\n",
    "            logits = model(x)\n",
    "            # logits: [b, 10]\n",
    "            # label:  [b]\n",
    "            # loss: tensor scalar\n",
    "            loss = criteon(logits, label)\n",
    "\n",
    "            # backprop\n",
    "\n",
    "            loss.backward()\n",
    "            optimizer.step()\n",
    "            optimizer.zero_grad()\n",
    "\n",
    "\n",
    "        print(epoch, 'loss:', loss.item())\n",
    "\n",
    "\n",
    "        model.eval()\n",
    "        with torch.no_grad():\n",
    "            # test\n",
    "            total_correct = 0\n",
    "            total_num = 0\n",
    "            for x, label in cifar_test:\n",
    "                # [b, 3, 32, 32]\n",
    "                # [b]\n",
    "                x, label = x.to(device), label.to(device)\n",
    "\n",
    "                # [b, 10]\n",
    "                logits = model(x)\n",
    "                # [b]\n",
    "                pred = logits.argmax(dim=1)\n",
    "                # [b] vs [b] => scalar tensor\n",
    "                correct = torch.eq(pred, label).float().sum().item()\n",
    "                total_correct += correct\n",
    "                total_num += x.size(0)\n",
    "                # print(correct)\n",
    "\n",
    "            # 计算准确率\n",
    "            acc = total_correct / total_num\n",
    "            # 打印当前轮次的准确率\n",
    "            print(epoch, 'test acc:', acc)\n",
    "\n",
    "main()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1 1.251025915145874 0.8125\n",
      "2 0.7969766855239868 0.8125\n",
      "3 0.5991461873054504 0.875\n",
      "4 0.4620618224143982 0.9375\n",
      "5 0.3615122139453888 1.0\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "import torchvision.datasets as dsets\n",
    "import torchvision.transforms as transforms\n",
    "\n",
    "\n",
    "# 设置随机种子\n",
    "torch.manual_seed(777)\n",
    "\n",
    "# 设置训练轮数和批量大小\n",
    "epochs=5\n",
    "batch_size=64\n",
    "# 加载CIFAR10训练集和测试集\n",
    "cifar10_train=dsets.CIFAR10('..\\..\\DATA\\CIFAR10_data',train=True,transform=transforms.ToTensor())\n",
    "cifar10_test=dsets.CIFAR10('..\\..\\DATA\\CIFAR10_data',train=False,transform=transforms.ToTensor())\n",
    "\n",
    "# 加载训练集数据\n",
    "data_loader=torch.utils.data.DataLoader(dataset=cifar10_train,batch_size=batch_size,shuffle=True)\n",
    "\n",
    "# 设置设备\n",
    "device=torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')\n",
    "\n",
    "# 定义卷积神经网络\n",
    "class Cnn(torch.nn.Module):\n",
    "    def __init__(self):\n",
    "        super(Cnn, self).__init__()\n",
    "        self._build_net()\n",
    "\n",
    "    def _build_net(self):\n",
    "\n",
    "        # 定义第一层卷积神经网络\n",
    "        self.layer1=torch.nn.Sequential(\n",
    "            torch.nn.Conv2d(3,64,kernel_size=3,stride=1,padding=1),\n",
    "            torch.nn.BatchNorm2d(64),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(64, 64, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(64),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.MaxPool2d(kernel_size=(2,2),stride=2),\n",
    "        )\n",
    "\n",
    "        # 定义第二层卷积神经网络\n",
    "        self.layer2 = torch.nn.Sequential(\n",
    "            torch.nn.Conv2d(64, 128, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(128),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(128, 128, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(128),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.MaxPool2d(kernel_size=(2, 2), stride=2),\n",
    "        )\n",
    "        # 定义第三层卷积神经网络\n",
    "        self.layer3 = torch.nn.Sequential(\n",
    "            torch.nn.Conv2d(128, 256, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(256),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(256),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(256),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.MaxPool2d(kernel_size=(2, 2), stride=2),\n",
    "        )\n",
    "        # 定义第四层卷积神经网络\n",
    "        self.layer4 = torch.nn.Sequential(\n",
    "            torch.nn.Conv2d(256, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.MaxPool2d(kernel_size=(2, 2), stride=2),\n",
    "        )\n",
    "        # 定义第五层卷积神经网络\n",
    "        self.layer5 = torch.nn.Sequential(\n",
    "            torch.nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),\n",
    "            torch.nn.BatchNorm2d(512),\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.MaxPool2d(kernel_size=(2, 2), stride=2),\n",
    "        )\n",
    "        # self.fc=torch.nn.flatten()\n",
    "        # 定义全连接层，将输入展平为一维向量\n",
    "        self.fc1=torch.nn.Linear(1*1*512,4096,bias=True)\n",
    "        # 定义第一个全连接层，输入维度为1*1*512，输出维度为4096，使用偏置\n",
    "        self.layer6=torch.nn.Sequential(\n",
    "            self.fc1,\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Dropout(0.5)\n",
    "        )\n",
    "        # 定义一个包含全连接层、ReLU激活函数和Dropout层的序列\n",
    "        self.fc2 = torch.nn.Linear(4096, 1000, bias=True)\n",
    "        # 定义第二个全连接层，输入维度为4096，输出维度为1000，使用偏置\n",
    "        self.layer7 = torch.nn.Sequential(\n",
    "            self.fc2,\n",
    "            torch.nn.ReLU(),\n",
    "            torch.nn.Dropout(0.5)\n",
    "        )\n",
    "        # 定义一个包含全连接层、ReLU激活函数和Dropout层的序列\n",
    "        self.fc3 = torch.nn.Linear(1000, 10, bias=True)\n",
    "        # 定义第三个全连接层，输入维度为1000，输出维度为10，使用偏置\n",
    "        self.criterion=torch.nn.CrossEntropyLoss()\n",
    "        # 定义交叉熵损失函数\n",
    "        self.optimizer=torch.optim.Adam(self.parameters(),lr=0.0001)\n",
    "\n",
    "        # 定义Adam优化器，学习率为0.0001\n",
    "    def forward(self,x):\n",
    "        # 前向传播\n",
    "        out=self.layer1(x)\n",
    "        out=self.layer2(out)\n",
    "        out=self.layer3(out)\n",
    "        out=self.layer4(out)\n",
    "        out=self.layer5(out)\n",
    "        out=out.view(out.size()[0],-1)\n",
    "        # out=self.fc(out)\n",
    "        out=self.fc1(out)\n",
    "        out=self.fc2(out)\n",
    "        out=self.fc3(out)\n",
    "        return out\n",
    "\n",
    "    def predict(self,x):\n",
    "        # 预测\n",
    "        self.eval()\n",
    "        return self.forward(x)\n",
    "\n",
    "    def get_accuracy(self,x,y):\n",
    "        # 获取准确率\n",
    "        prediction=self.predict(x)\n",
    "        corect_predict=(torch.max(prediction.data,1)[1]==y.data)\n",
    "        self.accuracy=corect_predict.float().mean()\n",
    "        return self.accuracy\n",
    "    def train_model(self,x,y):\n",
    "        self.train()#模型 定义中的 drop 会使用，批处理归一化 BatchNomalization也会使用\n",
    "        h=self.forward(x)\n",
    "        self.cost=self.criterion(h,y)\n",
    "        self.cost.backward()\n",
    "        self.optimizer.step()\n",
    "        self.optimizer.zero_grad()\n",
    "        return self.cost\n",
    "\n",
    "# 定义模型，并将其移动到指定设备上\n",
    "model=Cnn().to(device)\n",
    "\n",
    "# 遍历所有训练轮次\n",
    "for i in range(epochs):\n",
    "    avg=0\n",
    "    # 计算每个训练轮次的批次数量\n",
    "    total_batch=len(cifar10_train)//batch_size\n",
    "    # 遍历所有批次\n",
    "    for j,(batch_x,batch_y) in enumerate(data_loader):\n",
    "        # 将批次数据移动到指定设备上\n",
    "        X=torch.autograd.Variable(batch_x).cuda()\n",
    "        Y=torch.autograd.Variable(batch_y).cuda()\n",
    "\n",
    "        # 训练模型，并获取损失值\n",
    "        cost=model.train_model(X,Y)\n",
    "\n",
    "        # 累加损失值\n",
    "        avg+=cost.data/total_batch\n",
    "\n",
    "        # 获取模型在当前批次上的准确率\n",
    "        acc=model.get_accuracy(X,Y)\n",
    "\n",
    "    # 打印当前训练轮次的损失值和准确率\n",
    "    print(i+1,avg.item(),acc.item())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "import torch\n",
    "from torch import nn\n",
    "import torch.nn.functional as F\n",
    "\n",
    "\n",
    "# 首先定义一个包含conv与ReLu的基础卷积类\n",
    "class BasicConv2d(nn.Module):\n",
    "    def __init__(self, in_channels, out_channels, kernel_size, padding=0):\n",
    "        super(BasicConv2d, self).__init__()\n",
    "        # 定义一个卷积层，输入通道数为in_channels，输出通道数为out_channels，卷积核大小为kernel_size，padding为padding\n",
    "        self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, padding=padding)\n",
    "\n",
    "    def forward(self, x):\n",
    "        # 对输入x进行卷积操作，然后使用ReLU激活函数\n",
    "        x = self.conv(x)\n",
    "        return F.relu(x, inplace=True)\n",
    "\n",
    "\n",
    "# Inceptionv1的类,初始化的时候需要提供各个子模块的通道数大小\n",
    "class Inceptionv1(nn.Module):\n",
    "    def __init__(self, in_dim=3, hid_1_1=64, hid_2_1=96, hid_2_3=128, hid_3_1=16, out_3_5=32, out_4_1=32):\n",
    "        # 默认以这种形式创建 Inceptionv1(3, 64, 96, 128, 16, 32, 32)\n",
    "        super(Inceptionv1, self).__init__()\n",
    "        # 下面是4个子模块各自的网络定义\n",
    "        self.branch1x1 = BasicConv2d(in_dim, hid_1_1, 1)\n",
    "        self.branch3x3 = nn.Sequential(\n",
    "            BasicConv2d(in_dim, hid_2_1, 1),\n",
    "            BasicConv2d(hid_2_1, hid_2_3, 3, padding=1)\n",
    "        )\n",
    "        self.brach5x5 = nn.Sequential(\n",
    "            BasicConv2d(in_dim, hid_3_1, 1),\n",
    "            BasicConv2d(hid_3_1, out_3_5, 5, padding=2)\n",
    "\n",
    "        )\n",
    "        self.brach_pool = nn.Sequential(\n",
    "            nn.MaxPool2d(3, stride=1, padding=1),\n",
    "            BasicConv2d(in_dim, out_4_1, 1)\n",
    "        )\n",
    "        self.outlayer = nn.Linear((64+128+128+32) * 1 * 1, 10)\n",
    "\n",
    "    def forward(self, x):\n",
    "        # 对输入x进行第一个分支的处理，输出通道数为64\n",
    "        b1 = self.branch1x1(x)#64\n",
    "        # 对输入x进行第二个分支的处理，输出通道数为128\n",
    "        b2 = self.branch3x3(x)#128\n",
    "        # 对输入x进行第三个分支的处理，输出通道数为128\n",
    "        b3 = self.branch3x3(x)#128\n",
    "        # 对输入x进行第四个分支的处理，输出通道数为32\n",
    "        b4 = self.brach_pool(x)#32\n",
    "        # 将这四个模块沿着通道方向进行拼接\n",
    "        output_inc = torch.cat((b1, b2, b3, b4), dim=1)\n",
    "        # 对拼接后的结果进行自适应平均池化，输出尺寸为1x1\n",
    "        out_avg = F.adaptive_avg_pool2d(output_inc, [1, 1])\n",
    "        # 打印输出尺寸\n",
    "        print(out_avg.size())\n",
    "        # 将输出尺寸展平为一维\n",
    "        out_avg = out_avg.view(out_avg.size(0), -1)\n",
    "        # 打印展平后的输出尺寸\n",
    "        print(\"_______________\")\n",
    "        print(out_avg.size())\n",
    "        print(\"_______________\")\n",
    "        # 对展平后的输出进行全连接层的处理\n",
    "        output = self.outlayer(out_avg)\n",
    "        # 返回输出\n",
    "        return output\n",
    "\n",
    "\n",
    "\n",
    "module=Inceptionv1(192,64,96,128,16,32,32)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Inception2(\n",
      "  (branch1): BasicConv2d(\n",
      "    (conv): Conv2d(192, 96, kernel_size=(1, 1), stride=(1, 1))\n",
      "    (bn): BatchNorm2d(96, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "  )\n",
      "  (branch2): Sequential(\n",
      "    (0): BasicConv2d(\n",
      "      (conv): Conv2d(192, 48, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(48, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(48, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))\n",
      "      (bn): BatchNorm2d(64, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      "  (branch3): Sequential(\n",
      "    (0): BasicConv2d(\n",
      "      (conv): Conv2d(192, 64, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(64, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(64, 96, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))\n",
      "      (bn): BatchNorm2d(96, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "    (2): BasicConv2d(\n",
      "      (conv): Conv2d(96, 96, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))\n",
      "      (bn): BatchNorm2d(96, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      "  (branch4): Sequential(\n",
      "    (0): AvgPool2d(kernel_size=3, stride=1, padding=1)\n",
      "    (1): BasicConv2d(\n",
      "      (conv): Conv2d(192, 64, kernel_size=(1, 1), stride=(1, 1))\n",
      "      (bn): BatchNorm2d(64, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)\n",
      "    )\n",
      "  )\n",
      ")\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "from torch import nn\n",
    "import torch.nn.functional as F\n",
    "\n",
    "\n",
    "# 构建基础的卷积模块，与Inception V2的基础模块比，增加了BN层\n",
    "class BasicConv2d(nn.Module):\n",
    "    # 定义一个基本的二维卷积层\n",
    "    def __init__(self, in_channels, out_channels, kernel_size, padding=0):\n",
    "        # 初始化函数，传入输入通道数、输出通道数、卷积核大小、填充大小\n",
    "        super(BasicConv2d, self).__init__()\n",
    "        # 调用父类的初始化函数\n",
    "        self.conv = nn.Conv2d(in_channels, out_channels, kernel_size, padding=padding)\n",
    "        # 定义一个二维卷积层，传入输入通道数、输出通道数、卷积核大小、填充大小\n",
    "        self.bn = nn.BatchNorm2d(out_channels, eps=0.001)\n",
    "\n",
    "        # 定义一个批量归一化层，传入输出通道数、epsilon值\n",
    "    def forward(self, x):\n",
    "        # 定义前向传播函数，传入输入数据\n",
    "        x = self.conv(x)\n",
    "        # 将输入数据传入卷积层\n",
    "        x = self.bn(x)\n",
    "        # 将卷积层输出传入批量归一化层\n",
    "        return F.relu(x, inplace=True)\n",
    "\n",
    "\n",
    "class Inception2(nn.Module):\n",
    "    def __init__(self):\n",
    "        super(Inception2, self).__init__()\n",
    "        # 具体对应如Inception v2网络结构图（上图）\n",
    "        # 对应1x1卷积分支\n",
    "        self.branch1 = BasicConv2d(192, 96, 1, 0)\n",
    "        # 对应1x1卷积与3x3卷积分支\n",
    "        self.branch2 = nn.Sequential(\n",
    "            BasicConv2d(192, 48, 1, 0),\n",
    "            BasicConv2d(48, 64, 3, 1)\n",
    "        )\n",
    "        # 对应1x1卷积，3x3卷积，3x3卷积\n",
    "        self.branch3 = nn.Sequential(\n",
    "            BasicConv2d(192, 64, 1, 0),\n",
    "            BasicConv2d(64, 96, 3, 1),\n",
    "            BasicConv2d(96, 96, 3, 1)\n",
    "        )\n",
    "        # 对应3x3平均池化和1x1卷积\n",
    "        self.branch4 = nn.Sequential(\n",
    "            nn.AvgPool2d(3, stride=1, padding=1, count_include_pad=False),\n",
    "            BasicConv2d(192, 64, 1, 0)\n",
    "        )\n",
    "\n",
    "    # 前向过程\n",
    "    def forward(self, x):\n",
    "        # 将输入x传入branch1，得到x0\n",
    "        x0 = self.branch1(x)\n",
    "        x1 = self.branch2(x)  # 将输入x经过branch2分支处理\n",
    "        x2 = self.branch3(x)  # 将输入x经过branch3分支处理\n",
    "        x3 = self.branch4(x)  # 将输入x经过branch4分支处理\n",
    "        out = torch.cat((x0, x1, x2, x3), 1)  # 将四个分支处理的结果在通道维度上拼接起来\n",
    "        return out  # 返回拼接后的结果\n",
    "\n",
    "\n",
    "# 创建一个Inception2模块的实例\n",
    "module=Inception2()\n",
    "# 打印Inception2模块的实例\n",
    "print(module)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "block: torch.Size([2, 128, 8, 8])\n",
      "resnet: torch.Size([2, 10])\n"
     ]
    }
   ],
   "source": [
    "import  torch\n",
    "from    torch import  nn\n",
    "from    torch.nn import functional as F\n",
    "\n",
    "class ResBlk(nn.Module):\n",
    "    \"\"\"\n",
    "    resnet block\n",
    "    \"\"\"\n",
    "    def __init__(self, ch_in, ch_out, stride=1):\n",
    "        \"\"\"\n",
    "\n",
    "        :param ch_in:\n",
    "        :param ch_out:\n",
    "        \"\"\"\n",
    "        super(ResBlk, self).__init__()\n",
    "\n",
    "        # we add stride support for resbok, which is distinct from tutorials.\n",
    "        self.conv1 = nn.Conv2d(ch_in, ch_out, kernel_size=3, stride=stride, padding=1)\n",
    "        self.bn1 = nn.BatchNorm2d(ch_out)\n",
    "        self.conv2 = nn.Conv2d(ch_out, ch_out, kernel_size=3, stride=1, padding=1)\n",
    "        self.bn2 = nn.BatchNorm2d(ch_out)\n",
    "\n",
    "        self.extra = nn.Sequential()\n",
    "        if ch_out != ch_in:\n",
    "            # [b, ch_in, h, w] => [b, ch_out, h, w]\n",
    "            self.extra = nn.Sequential(\n",
    "                nn.Conv2d(ch_in, ch_out, kernel_size=1, stride=stride),\n",
    "                nn.BatchNorm2d(ch_out)\n",
    "            )\n",
    "\n",
    "\n",
    "    def forward(self, x):\n",
    "        \"\"\"\n",
    "\n",
    "        :param x: [b, ch, h, w]\n",
    "        :return:\n",
    "        \"\"\"\n",
    "        # 对输入进行卷积操作，并使用批归一化\n",
    "        out = F.relu(self.bn1(self.conv1(x)))\n",
    "        # 对卷积后的结果进行卷积操作，并使用批归一化\n",
    "        out = self.bn2(self.conv2(out))\n",
    "        # short cut.\n",
    "        # extra module: [b, ch_in, h, w] => [b, ch_out, h, w]\n",
    "        # element-wise add:\n",
    "        # 对输入进行额外的卷积操作，并将结果与卷积后的结果进行元素级相加\n",
    "        out = self.extra(x) + out\n",
    "        # 对相加后的结果进行ReLU激活\n",
    "        out = F.relu(out)\n",
    "        \n",
    "        return out\n",
    "\n",
    "class ResNet18(nn.Module):\n",
    "\n",
    "    def __init__(self):\n",
    "        super(ResNet18, self).__init__()\n",
    "\n",
    "        # 32 * 32 * 3 --> 10 * 10 * 64\n",
    "        self.conv1 = nn.Sequential(\n",
    "            nn.Conv2d(3, 64, kernel_size=3, stride=3, padding=0),\n",
    "            nn.BatchNorm2d(64)\n",
    "        )\n",
    "        # followed 4 blocks\n",
    "        # [b, 64, h, w] => [b, 128, h ,w]\n",
    "        self.blk1 = ResBlk(64, 128, stride=2) # 5 * 5 * 128\n",
    "        # [b, 128, h, w] => [b, 256, h, w]\n",
    "        self.blk2 = ResBlk(128, 256, stride=2) # 3 * 3 * 256\n",
    "        # # [b, 256, h, w] => [b, 512, h, w]\n",
    "        self.blk3 = ResBlk(256, 512, stride=2) # 2 * 2 * 512\n",
    "        # # [b, 512, h, w] => [b, 1024, h, w]\n",
    "        self.blk4 = ResBlk(512, 512, stride=2) # 1 * 1 * 512\n",
    "\n",
    "        self.outlayer = nn.Linear(512*1*1, 10)\n",
    "\n",
    "    def forward(self, x):\n",
    "        \"\"\"\n",
    "\n",
    "        :param x:\n",
    "        :return:\n",
    "        \"\"\"\n",
    "        x = F.relu(self.conv1(x))\n",
    "\n",
    "        # [b, 64, h, w] => [b, 1024, h, w]\n",
    "        x = self.blk1(x)\n",
    "        x = self.blk2(x)\n",
    "        x = self.blk3(x)\n",
    "        x = self.blk4(x)\n",
    "\n",
    "\n",
    "        # print('after conv:', x.shape) #[b, 512, 2, 2]\n",
    "        # [b, 512, h, w] => [b, 512, 1, 1]\n",
    "        x = F.adaptive_avg_pool2d(x, [1, 1])\n",
    "        # print('after pool:', x.shape)\n",
    "        x = x.view(x.size(0), -1)\n",
    "        x = self.outlayer(x)\n",
    "        return x\n",
    "\n",
    "\n",
    "# 验证网络是否正确， 测试网络\n",
    "blk = ResBlk(64, 128, stride=4)\n",
    "tmp = torch.randn(2, 64, 32, 32)\n",
    "out = blk(tmp)\n",
    "print('block:', out.shape)\n",
    "\n",
    "x = torch.randn(2, 3, 32, 32)\n",
    "model = ResNet18()\n",
    "out = model(x)\n",
    "print('resnet:', out.shape)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
